# bash run.sh

# 训练数据准备
#python offline_texture_swapping.py \
#--dataroot "/root/autodl-fs/Make-It-3D-master/results/test5/train"

# 训练10个epoch
#python train.py \
#--n_epochs 3 \
#--use_weights \
#--dataroot "/root/autodl-fs/Make-It-3D-master/results/test5/train" \
#--netG_pre "/root/autodl-fs/Make-It-3D-master/weights/srntt_weights/netG_100.pth" \
#--netD_pre "/root/autodl-fs/Make-It-3D-master/weights/srntt_weights/netD_100.pth"

# 推理(vgg的路径需要在miniconda3/lib/python3.8/site-packages/torch/hub.py中配置)
#--weight "/root/autodl-fs/Make-It-3D-master/weights/srntt_weights/netG_100.pth" \
python online_inference.py \
--save_path './out3.png' \
--input "/root/autodl-fs/Make-It-3D-master/results/test5/train/2700.png" \
--ref "/root/autodl-fs/Make-It-3D-master/results/test5/train/2700_gt.png" \
--weight "/root/autodl-fs/Make-It-3D-master/srntt/weights/netG_003.pth" \
--use_weights